hyp = {
    'lr0': 0.01, 
    'lrf': 0.01, 
    'momentum': 0.937, 
    'weight_decay': 0.0005, 
    'warmup_epochs': 3.0, 
    'warmup_momentum': 0.8, 
    'warmup_bias_lr': 0.1, 
    'box': 0.05, 
    'cls': 0.5, 
    'cls_pw': 1.0, 
    'obj': 1.0, 
    'obj_pw': 1.0, 
    'iou_t': 0.2, 
    'anchor_t': 4.0, 
    'fl_gamma': 0.0,
    'hsv_h': 0.015,
    'hsv_s': 0.7,
    'hsv_v': 0.4,
    'degrees': 0.0,
    'translate': 0.1,
    'scale': 0.5,
    'shear': 0.0,
    'perspective': 0.0,
    'flipud': 0.0,
    'fliplr': 0.5,
    'mosaic': 1.0, 
    'mixup': 0.0,
    'copy_paste': 0.0,
}
configs = {
    "classification": {
        "yolov5n-cls": {
            "params": "0.5 M",
            "flops_224x224": "2.5 GFLOPs"
        },
        "yolov5-efficientnet_b0": {
            "params": "1.0 M",
            "flops_224x224": "5.3 GFLOPs"
        },
        "yolov5s-cls": {
            "params": "1.4 M",
            "flops_224x224": "5.4 GFLOPs"
        },
        "yolov5-efficientnet_b1": {
            "params": "1.5 M",
            "flops_224x224": "7.8 GFLOPs"
        },
        "yolov5-efficientnet_b2": {
            "params": "1.7 M",
            "flops_224x224": "9.1 GFLOPs"
        },
        "yolov5-resnet18": {
            "params": "3.7 M",
            "flops_224x224": "11.7 GFLOPs"
        },
        "yolov5-efficientnet_b3": {
            "params": "2.4 M",
            "flops_224x224": "12.2 GFLOPs"
        },
        "yolov5m-cls": {
            "params": "3.9 M",
            "flops_224x224": "12.9 GFLOPs"
        },
        "yolov5-resnet34": {
            "params": "7.4 M",
            "flops_224x224": "21.8 GFLOPs"
        },
        "yolov5-resnet50": {
            "params": "8.5 M",
            "flops_224x224": "25.6 GFLOPs"
        },
        "yolov5l-cls": {
            "params": "8.5 M",
            "flops_224x224": "26.5 GFLOPs"
        },
        "yolov5-resnet101": {
            "params": "15.9 M",
            "flops_224x224": "44.5 GFLOPs"
        },
        "yolov5x-cls": {
            "params": "15.9 M",
            "flops_224x224": "48.1 GFLOPs"
        },
    },
    "detection": {
        "yolov5n": {
            "params": "4.5 M",
            "flops_640x640": "1.9 GFLOPs"
        },
        "yolov5n6": {
            "params": "4.6 M",
            "flops_640x640": "3.2 GFLOPs"
        },
        "yolov5s": {
            "params": "16.5 M",
            "flops_640x640": "7.2 GFLOPs"
        },
        "yolov5s6": {
            "params": "16.8 M",
            "flops_640x640": "12.6 GFLOPs"
        },
        "yolov5m": {
            "params": "49.0 M",
            "flops_640x640": "21.2 GFLOPs"
        },
        "yolov5m6": {
            "params": "50.0 M",
            "flops_640x640": "35.7 GFLOPs"
        },
        "yolov5l": {
            "params": "109.1 M",
            "flops_640x640": "46.5 GFLOPs"
        },
        "yolov5l6": {
            "params": "111.4 M",
            "flops_640x640": "76.8 GFLOPs"
        },
        "yolov5x": {
            "params": "205.7 M",
            "flops_640x640": "86.7 GFLOPs"
        },
        "yolov5x6": {
            "params": "209.8 M",
            "flops_640x640": "140.7 GFLOPs"
        }
    },
    "segmentation": {
        "yolov5n-seg": {
            "params": "2.0 M",
            "flops_640x640": "7.1 GFLOPs"
        },
        "yolov5s-seg": {
            "params": "1.4 M",
            "flops_640x640": "5.4 GFLOPs"
        },
        "yolov5m-seg": {
            "params": "3.9 M",
            "flops_640x640": "12.9 GFLOPs"
        },
        "yolov5l-seg": {
            "params": "8.5 M",
            "flops_640x640": "26.5 GFLOPs"
        },
        "yolov5x-seg": {
            "params": "15.9 M",
            "flops_640x640": "48.1 GFLOPs"
        },
    },
}